
from datasets.dataloader import BaseDataset, TemporalDataset


def lidarcap_data_generator(options, is_train=True, is_use_image=False, is_aug=False, is_visual=False):
    data = BaseDataset(options=options, dataset='lidarcap', is_train=is_train,
                       is_use_image=is_use_image, is_aug=is_aug, is_visual=is_visual)
    return data


def lidarcap_seq_data_generator(options, dataset, is_train=True, is_use_image=False, is_aug=False, is_visual=False, need_depth=False):
    return TemporalDataset(dataset=dataset, is_train=is_train, is_visual=is_visual, need_depth=need_depth)


# def lidarcao_seq_reduced_data_generator(options, is_train=True, is_use_image=False, is_aug=False, is_visual=False):
#     return TemporalDataset(dataset='lidarcap_seq_reduced', is_train=is_train)
